#!/usr/bin/env python
# coding:utf-8

import requests
from hashlib import md5

class Chaojiying_Client(object):

    def __init__(self, username, password, soft_id):
        self.username = username
        password =  password.encode('utf8')
        self.password = md5(password).hexdigest()
        self.soft_id = soft_id
        self.base_params = {
            'user': self.username,
            'pass2': self.password,
            'softid': self.soft_id,
        }
        self.headers = {
            'Connection': 'Keep-Alive',
            'User-Agent': 'Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 5.1; Trident/4.0)',
        }

    def PostPic(self, im, codetype):
        """
        im: 图片字节
        codetype: 题目类型 参考 http://www.chaojiying.com/price.html
        """
        params = {
            'codetype': codetype,
        }
        params.update(self.base_params)
        files = {'userfile': ('ccc.jpg', im)}
        r = requests.post('http://upload.chaojiying.net/Upload/Processing.php', data=params, files=files, headers=self.headers)
        return r.json()

    def PostPic_base64(self, base64_str, codetype):
        """
        im: 图片字节
        codetype: 题目类型 参考 http://www.chaojiying.com/price.html
        """
        params = {
            'codetype': codetype,
            'file_base64':base64_str
        }
        params.update(self.base_params)
        r = requests.post('http://upload.chaojiying.net/Upload/Processing.php', data=params, headers=self.headers)
        return r.json()

    def ReportError(self, im_id):
        """
        im_id:报错题目的图片ID
        """
        params = {
            'id': im_id,
        }
        params.update(self.base_params)
        r = requests.post('http://upload.chaojiying.net/Upload/ReportError.php', data=params, headers=self.headers)
        return r.json()


def setSessionHeaders(session, items):
    """统一添加请求头信息"""
    for k, v in items.items():
        session.headers[k] = v

import base64
def supervisesCode():
    url = "https://apif.wozhipei.com/supervises/code"
    session = requests.session()
    setSessionHeaders(session ,{
        'User-Agent': "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/130.0.0.0 Safari/537.36",
        'Authorization': 'Bearer eyJhbGciOiJIUzM4NCJ9.eyJqdGkiOiJqd3RfaWQiLCJzdWIiOiIxNzI5NjEyMEIxSEZZWEE4IiwiaWF0IjoxNzMxMzg0OTk2LCJhdWQiOiJXRUIiLCJleHAiOjE3MzE5OTAwOTYsInNpZCI6MTAxOCwidHlwZSI6IlVTRVIiLCJncm91cFR5cGUiOjF9.vo47fhxJRK8DZEkX6WGKLsFHPu_s7m5YYZW-U3_82_YrnMz4plemn_mrpuqVl-HH'
    })
    headers = {
        "Host": "apif.wozhipei.com",
        "Origin": "https://admin-jiangxi.zhipeizaixian.com",
        "X-User-Type": "1",
    }
    newHeaders = {**session.headers, **headers}
    response = session.get(url, headers=newHeaders)
    base64_str = response.content.decode("utf-8").split(",")[1]
    img_data = base64.b64decode(base64_str)
    with open('code.png', 'wb') as f:
        f.write(img_data)

if __name__ == '__main__':
    supervisesCode()
    chaojiying = Chaojiying_Client('18279751024', 'gwl12344321', '964905')	#用户中心>>软件ID 生成一个替换 96001
    im = open('code.png', 'rb').read()													#本地图片文件路径 来替换 a.jpg 有时WIN系统须要//
    # response = chaojiying.PostPic(im, 6001)												#1902 验证码类型  官方网站>>价格体系 3.4+版 print 后要加()
    # pic_id = response['pic_id']
    # pic_str = response['pic_str']
    # err_str = response['err_str']
    # err_no = response['err_no']
    # print(chaojiying.PostPic_base64(base64_str, 6004))  #此处为传入 base64代码
